from src.yaml_util import Yaml


class Config(object):

    def __init__(self, file_path="conf/config.yml"):
        self.yaml = Yaml(file_path)

    def get_conf(self, key):
        return self.yaml.read_yml(key)

    def set_conf(self, key, value):
        self.yaml.write_yml(key, value)
